03. 训练和记忆功能

03 训练和记忆功能 V1

递归层

PyTorch 中的主要递归层类型文档。请查看该文档并了解以下三大类型:RNN、LSTM 和 GRU。

隐藏状态维度

假设你定义了一个 GRU 层级,其中 input_size = 100hidden_size = 20 以及 num_layers=1
如果你一次传入 3 个序列的数据,并且批次优先,那么隐藏状态的维度是多少?

SOLUTION: `(1, 3, 20)`